writing automated tests with python